package com.cursordemo.controller;

import com.cursordemo.pojo.Result;
import com.cursordemo.service.FileService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

/**
 * 文件上传控制器
 */
@Slf4j
@RestController
public class FileController {
    
    @Autowired
    private FileService fileService;
    
    /**
     * 文件上传接口
     * @param file 上传的文件
     * @return 文件访问URL
     */
    @PostMapping("/upload")
    public Result<String> upload(@RequestParam("file") MultipartFile file) {
        log.info("文件上传请求 - 文件名: {}, 大小: {} bytes", 
                file.getOriginalFilename(), file.getSize());
        
        try {
            // 检查文件是否为空
            if (file.isEmpty()) {
                log.warn("上传的文件为空");
                return Result.error("上传的文件不能为空");
            }
            
            // 检查文件大小（限制为10MB）
            if (file.getSize() > 10 * 1024 * 1024) {
                log.warn("文件大小超过限制 - 文件名: {}, 大小: {} bytes", 
                        file.getOriginalFilename(), file.getSize());
                return Result.error("文件大小不能超过10MB");
            }
            
            // 检查文件类型
            String contentType = file.getContentType();
            if (contentType == null || !isAllowedFileType(contentType)) {
                log.warn("不支持的文件类型 - 文件名: {}, 类型: {}", 
                        file.getOriginalFilename(), contentType);
                return Result.error("不支持的文件类型");
            }
            
            // 上传文件到OSS
            String fileUrl = fileService.uploadFile(file);
            
            log.info("文件上传成功 - 文件名: {}, URL: {}", 
                    file.getOriginalFilename(), fileUrl);
            
            return Result.success(fileUrl);
            
        } catch (Exception e) {
            log.error("文件上传失败 - 文件名: {}", file.getOriginalFilename(), e);
            return Result.error("文件上传失败: " + e.getMessage());
        }
    }
    
    /**
     * 检查文件类型是否允许
     * @param contentType 文件类型
     * @return 是否允许
     */
    private boolean isAllowedFileType(String contentType) {
        // 允许的文件类型
        String[] allowedTypes = {
            "image/jpeg",
            "image/jpg", 
            "image/png",
            "image/gif",
            "image/webp",
            "application/pdf",
            "application/msword",
            "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
            "application/vnd.ms-excel",
            "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
            "text/plain"
        };
        
        for (String allowedType : allowedTypes) {
            if (allowedType.equals(contentType)) {
                return true;
            }
        }
        return false;
    }
}

